/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package br.com.tbp.gui.table;

import br.com.tbp.model.domain.Pedido;
import br.com.tbp.model.domain.StatusPedido;
import br.com.tbp.utils.DateUtils;
import java.util.ArrayList;
import java.util.List;
import javax.swing.table.AbstractTableModel;

/**
 *
 * @author Diego
 */
public class PedidosTableModel extends AbstractTableModel {

    private List<Pedido> lista;

    public PedidosTableModel(List<Pedido> lista) {
        this.lista = lista;
    }

    public PedidosTableModel() {
        this.lista = new ArrayList<Pedido>();
    }

    public int getRowCount() {
        return lista.size();
    }

    public int getColumnCount() {
        return 6;
    }

    @Override
    public Class<?> getColumnClass(int columnIndex) {
        switch (columnIndex) {
            case 0:
                return Long.class;
            case 1:
            case 2:
            case 3:
            case 4:
            case 5:
                return String.class;
        }
        return Object.class;
    }

    @Override
    public String getColumnName(int column) {
        switch (column) {
            case 0:
                return "ID pedido";
            case 1:
                return "Data emissão";
            case 2:
                return "Consultor";
            case 3:
                return "Empresa";
            case 4:
                return "CNPJ";
            case 5:
                return "Situação";
        }
        return null;
    }

    public Object getValueAt(int rowIndex, int columnIndex) {
        Pedido p = this.getPedido(rowIndex);
        if (p != null) {
            switch (columnIndex) {
                case 0:
                    return p.getId();
                case 1:
                    return DateUtils.dataObjetoParaString(p.getData());
                case 2:
                    return p.getConsultor().getNome();
                case 3:
                    return p.getEmpresa().getNomeFantasia();
                case 4:
                    return p.getEmpresa().getCnpj();
                case 5:
                    StatusPedido s = p.getSituacaoExternaAtual();
                    if(s != null) {
                        return s.toString();
                    }
                    return "";
            }
        }
        return null;
    }

    public Pedido getPedido(int row) {
        if (row >= 0) {
            return lista.get(row);
        }
        return null;
    }

    public boolean removePedido(Pedido c) {
        if (lista.remove(c)) {
            fireTableDataChanged();
            return true;
        }
        return false;
    }

    public List<Pedido> getListaDeItens() {
        return new ArrayList<Pedido>(lista);
    }
}
